package com.chris.code.controller;

import com.chris.code.model.Database;
import com.chris.code.service.DatabaseService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.sql.SQLException;

/**
 * Copyright (c) 2018-2025, chris All rights reserved.
 * <p>
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * <p>
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * Neither the name of the pig4cloud.com developer nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * <p>
 * Author: haoka (haokang207@126.com)
 * <p>
 * Date: 2018/9/30 10:11
 **/
@RestController
@RequestMapping("/table")
@Slf4j
public class TableController {
    @Autowired
    private DatabaseService databaseService;

    @PostMapping
    public Object getTables(Database database) {
        try {
            return databaseService.fetchTables(database);
        } catch (SQLException | ClassNotFoundException e) {
            log.error("database error", e);
            return e.getMessage();
        }
    }

    @PostMapping("/{tableName}")
    public Object getTableFields(@PathVariable("tableName") String tableName, Database database) {
        try {
            return databaseService.fetchTableField(database, tableName);
        } catch (SQLException | ClassNotFoundException e) {
            log.error("database error", e);
            return e.getMessage();
        }
    }
}
